

/**
 * 定义一个链表，并使用系统提供的方法操作链表
 */
public class MyLinkedList {

    private ListNode listNode;

    public void setListNode(ListNode listNode) {
        this.listNode = listNode;
    }

    public ListNode getListNode() {
        return listNode;
    }

    public MyLinkedList(ListNode head) {
        listNode = head;
    }

    public MyLinkedList() {

    }

    public int get(int index) {
        int size = size();
        if (index < 0 || index > size - 1) {
            return -1;
        }
        ListNode imghead = new ListNode();
        imghead.setVal(-1);
        imghead.setNext(listNode);

        ListNode cur = imghead.getNext();

        while (index > 0) {
            cur = cur.getNext();
            index = index - 1;
        }
        return cur.getVal(); //index=0也是返回cur本身，也就是第一个节点。 index不为0，最终的cur的值就是第index节点的值。
    }

    public ListNode getNode(int index) {
        int size = size();
        if (index < 0 || index > size - 1) {
            return null;
        }
        ListNode imghead = new ListNode();
        imghead.setVal(-1);
        imghead.setNext(listNode);

        ListNode cur = imghead.getNext();

        while (index > 0) {
            cur = cur.getNext();
            index = index - 1;
        }
        return cur; //index=0也是返回cur本身，也就是第一个节点。 index不为0，最终的cur的值就是第index节点的值。
    }

    public int size() {//获取长度
        ListNode cur = listNode;
        int size = 0;
        while (cur != null) {
            size = size + 1;
            cur = cur.getNext();
        }
        return size;
    }

    public void addAtHead(int val) {
        ListNode newNode = new ListNode();//定义要插入的节点
        newNode.setVal(val);//给要插入的节点赋值

        ListNode imghead = new ListNode();
        imghead.setVal(-1);
        imghead.setNext(listNode);//定义虚拟头节点

        ListNode cur = imghead.getNext();//当前节点移动到虚拟头节点的下一个节点

        imghead.setNext(null);//断开虚拟头节点与下一个节点连接
        imghead.setNext(newNode);//重新改变指向
        newNode.setNext(cur);//重新改变指向

        listNode = newNode;//最终符合题目要求的节点是newNode ，把newNode给listNode

    }

    //尾部插入节点
    public void addAtTail(int val) {


        ListNode newNode = new ListNode();//定义要插入的节点
        newNode.setVal(val);//给要插入的节点赋值

        ListNode imghead = new ListNode();
        imghead.setVal(-1);
        imghead.setNext(listNode);//定义虚拟头节点

        ListNode cur = imghead;//

        while (cur.getNext() != null) {
            cur = cur.getNext();
        }

        cur.setNext(newNode);

        listNode = imghead.getNext();

    }

    //在第index前节点插入一个节点
    public void addAtIndex(int index, int val) {
        int size = size();
        if (index < 0 || index > size) {
            System.out.println("传入的节点index小于0或index超过节点索引，无法插入数据");
        } else if(index==size){
            addAtTail(val);
        }
        else if (index == 0) {//在头节点前插入
            addAtHead(val);
        } else {//index从1到size-1之间
            ListNode newNode = new ListNode();//定义要插入的节点
            newNode.setVal(val);//给要插入的节点赋值

            ListNode imghead = new ListNode();
            imghead.setVal(-1);
            imghead.setNext(listNode);//定义虚拟头节点

            ListNode cur = imghead.getNext();

            while (index > 1) {//寻找要插入节点的前一个节点，因为要找的是前一个节点，所以要大于1
                cur = cur.getNext();
                index = index - 1;

            }

            ListNode tmp = cur.getNext();//先存下下一个节点

            cur.setNext(newNode);
            newNode.setNext(tmp);

        }


    }

    //删除第index的节点
    public void deleteAtIndex(int index) {
        int size = size();
        if (index < 0 || index > size - 1) {
            System.out.println("传入的节点index小于0或index超过节点索引，无法删除数据");
        } else {
            ListNode imghead = new ListNode();
            imghead.setVal(-1);
            imghead.setNext(listNode);//定义虚拟头节点


            if (index == 0) {//删除第一个节点
                ListNode tmp = imghead.getNext().getNext();
                imghead.setNext(null);
                imghead.setNext(tmp);
                listNode = imghead.getNext();
            } else {//index从1到size-1之间
                ListNode cur = imghead.getNext();
                while (index > 1) {//寻找要删除节点的前一个节点，因为要找的是前一个节点，所以要大于1
                    cur = cur.getNext();
                    index = index - 1;

                }

                ListNode tmp = cur.getNext().getNext();//先存下下一个节点

                cur.setNext(tmp);
                listNode = imghead.getNext();
            }


        }
    }
}

